from ship_envs.environment.single_ship_env import SingleShipEnv


def make_my_env():
    # callback = BaseCallback()
    env = SingleShipEnv()
    return env

if __name__ == '__main__':
    env = make_my_env()

    for _ in range(10):
        env.reset()
        env.render("")
        while True:
            obs, reward, done, info = env.step(env.action_space.sample())
            assert obs.shape == (3, 84, 84)
            env.render("")
            if done:
                break
